#ifndef MAINWINDOW_H
#define MAINWINDOW_H

#include <QMainWindow>
#include <opencv2/opencv.hpp>
#include <opencv2/dnn.hpp>
#include <QDebug>
#include <QTimer>

QT_BEGIN_NAMESPACE
namespace Ui { class MainWindow; }
QT_END_NAMESPACE

class MainWindow : public QMainWindow
{
    Q_OBJECT

public:
    MainWindow(QWidget *parent = nullptr);
    ~MainWindow();

     QImage detectHatInImage(const cv::Mat &image);
     bool detectHat(const cv::Mat& image, cv::dnn::Net& hat_net, std::vector<cv::Rect>& detectedBoxes,std::map<int, int>& detectedClasses);

private slots:
    void on_pushButton_clicked();
    void updateFrame();


    void on_pushButton_2_clicked();

    void on_comboBox_currentIndexChanged(int index);

private:
    Ui::MainWindow *ui;
    cv::VideoCapture *capture;  // OpenCV的VideoCapture对象
     QTimer *timer;  // 用于定时更新显示
 int frameCounter=0;
  cv::dnn::Net hat_net;
  int classId=0;
};
#endif // MAINWINDOW_H
